python lstm_fedma_with_comm.py